package demo.generator;

import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.engine.VelocityTemplateEngine;

import java.util.Map;

/**
 * 自定义实现的velocity模版引擎<br/>
 */
public class MyVelocityTemplateEngine extends VelocityTemplateEngine {

    @Override
    public Map<String, Object> getObjectMap(TableInfo tableInfo) {
        Map<String, Object> objectMap = super.getObjectMap(tableInfo);
        // 代码的生成日期固定为2017-01-01，避免因为这点问题而导致代码变更
        objectMap.put("date", "2017-01-01");
        return objectMap;
    }
}
